from models.molecule.unimap.unimap import UniMAP